#!/usr/bin/env python3
# _*_coding:utf-8_*_

"""
@version: 3.5.2
@author: cockroach 
@contact: donnie.fan@gmail.com
@software: PyCharm
@file: ex_code_xmlmod_4.py
@time: 2017/2/27 12:49
"""

import xml.etree.ElementTree as ET

file_tree = ET.parse("xmltest.xml")
root = file_tree.getroot()

'''
修改xml内容
取值需要修改变量类型为int
存值需要修改变量类型为str
.set 修改xml节点属性
'''

# 修改节点
for node in root.iter("year"):  # 如果节点是year
    year_new = int(node.text)+1 # 则：提取节点内容，并赋值给year_new变量
    print("----修改前")
    print(node.text)
    node.text = str(year_new)   # 修改节点内容
    print("----修改后")
    print(node.text)
    node.set("updated","yes")   # 修改节点属性

file_tree.write("xmltest1.xml")


'''
删除xml内容
.findall 查找指定的内容
'''

# 删除节点
for country_node in root.findall("country"):
    rank = int(country_node.find("rank").text)
    print(rank)
    if rank > 50:
        root.remove(country_node)

file_tree.write("output.xml")










